import os
import time
import requests
from jsonpath import jsonpath

def search_and_download_qq_music(query_text):

    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36 Edg/128.0.0.0',
        'Cookie': 'you cookie here'
    }

    # 请求的链接
    # query_text = 'Black Myth wukong'
    search_url = rf'https://u.y.qq.com/cgi-bin/musicu.fcg?data={{"comm":{{"g_tk":235530277,"uin":"1152921504916411742","format":"json","inCharset":"utf-8","outCharset":"utf-8","notice":0,"platform":"h5","needNewCode":1,"ct":23,"cv":0}},"req_0":{{"method":"DoSearchForQQMusicDesktop","module":"music.search.SearchCgiService","param":{{"remoteplace":"txt.mqq.all","searchid":"64237725668973550","search_type":0,"query":"{query_text}","page_num":1,"num_per_page":20}}}}}}'

    response = requests.get(search_url)
    music_list = jsonpath(response.json(), '$..data.body.song.list')[0]
    # print(f'music_list: {music_list}')

    if not os.path.exists('Q Music'):
        os.mkdir('Q Music')

    for item in music_list:
        music_mid = jsonpath(item, '$.mid')[0]
        music_name = jsonpath(item, '$.name')[0]

        # print(f'music_name: {music_name}, music_mid: {music_mid}')

        music_data_url = rf'https://c6.y.qq.com/splcloud/fcgi-bin/smartbox_new.fcg?data={{"comm":{{"cv":4747474,"ct":24,"format":"json","inCharset":"utf-8","outCharset":"utf-8","notice":0,"platform":"yqq.json","needNewCode":1,"uin":"1152921504916411742","g_tk_new_20200303":1849600344,"g_tk":1849600344}},"req_9":{{"module":"vkey.GetVkeyServer","method":"CgiGetVkey","param":{{"guid":"4868259520","songmid":["{music_mid}"],"songtype":[0],"uin":"1152921504916411742","loginflag":1,"platform":"20"}}}}}}'
        music_data_response = requests.get(music_data_url)
        # print(f'music_data_response: {music_data_response.json()}')

        data_info = jsonpath(music_data_response.json(), '$..purl')[0]
        # print(f'data_info: {data_info}')

        music_url = f'https://dl.stream.qqmusic.qq.com/{data_info}'

        music_response = requests.get(music_url, headers=headers)

        with open(f'./Q Music/{music_name}.mp3', 'wb') as file:
            file.write(music_response.content)

        print(f'《{music_name}》下载成功')
        print("-" * 20)
        time.sleep(0.5)

        # break

search_and_download_qq_music("Black Myth wukong")
